package vaani.shabd;

import edu.stanford.nlp.tagger.maxent.MaxentTagger;
import java.util.Vector;
import vaani.internal.WordTagPair;

/**
 *
 * @author sourcemorph
 */
public class Tagger {
    private static MaxentTagger tagger;

    static {
        try {
            tagger = new MaxentTagger("data/left3words-wsj-0-18.tagger");
        } catch(Exception ex) {
            ex.printStackTrace();
        }
    }

    public static Vector<WordTagPair> tag(String sentence) throws Exception {
        Vector<WordTagPair> tags = new Vector<WordTagPair>();
        String taggedSentence = tagger.tagString(sentence);
        String [] temp = taggedSentence.split(" ");
        for(int i=0; i<temp.length; i++) {
            String [] t = temp[i].split("/");
            WordTagPair w = new WordTagPair(t[0], t[1]);
            tags.add(w);
        }
        return tags;
    }
}
